package DemoReview;
/**
 * 和可被 k 整除的子数组
 */

import java.util.HashMap;
import java.util.Map;
public class LeetCode974 {
    public int subarraysDivByK(int[] nums, int k) {
        Map<Integer, Integer> hash = new HashMap<>();
        hash.put(0 % k, 1);
        int sum = 0;
        int ret = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            int tmp = (sum % k + k) % k;
            ret += hash.getOrDefault(tmp, 0);
            hash.put(tmp, hash.getOrDefault(tmp, 0) + 1);
        }
        return ret;
    }
}
